/**
 * ImportExcelUtils.java
 * Created at 2017-12-20
 * Created by zhanghailin
 * Copyright (C) 2017 BROADTEXT SOFTWARE, All rights reserved.
 */
package com.wisdytech.common.utils;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.*;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.List;

/**
 * <p>ClassName: ImportExcelUtils</p>
 * <p>Description: TODO</p>
 * <p>Author: zhanghailin</p>
 * <p>Date: 2017-12-20</p>
 */
public class ImportExcelUtils {
    
    private final static String excel2003L =".xls";    //2003- 版本的excel
    private final static String excel2007U =".xlsx";   //2007+ 版本的excel
      
    /** 
     * 描述：获取IO流中的数据，组装成List<List<Object>>对象 
     * @param in 输入流
     * @param fileName 文件名
     * @return list对象
     */
    public  static List<List<Object>> getDataListByExcel(InputStream in, String fileName) {
        List<List<Object>> list = null;
        //创建Excel工作薄
        Workbook work;
        try {
            work = getWorkbook(in,fileName);
            if(null == work){  
                throw new Exception("Excel工作薄为空！");
            } 
            Sheet sheet;
            Row row;
            Cell cell;
            list = new ArrayList<>();
            //遍历Excel中所有的sheet  
            for (int i = 0; i < work.getNumberOfSheets(); i++) {  
                sheet = work.getSheetAt(i);  
                if(sheet==null){continue;}  
                //遍历当前sheet中的所有行(保含表头)
                for (int j = 0; j <= sheet.getPhysicalNumberOfRows(); j++) {
                    row = sheet.getRow(j);  
                    if(row==null){continue;}  
                    //遍历所有的列
                    List<Object> li = new ArrayList<>();
                    for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {  
                        cell = row.getCell(y);
                        if (cell ==null) {
                            li.add("");
                        } else {
                            li.add(getCellValue(cell));
                        }
                    }
                    list.add(li);  
                }  
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if(in!=null){
                    in.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
            
        }
        return list;  
    }  
      
    /** 
     * 描述：根据文件后缀，自适应上传文件的版本  
     * @param inStr 输入流
     * @param fileName 文件名
     * @return Workbook
     * @throws Exception 异常
     */  
    public static Workbook getWorkbook(InputStream inStr, String fileName) throws Exception {
        Workbook wb = null;
        String fileType = fileName.substring(fileName.lastIndexOf("."));
        if(excel2003L.equals(fileType)){  
            wb = new HSSFWorkbook(inStr);  //2003-
        }else if(excel2007U.equals(fileType)){  
            wb = new XSSFWorkbook(inStr);  //2007+
        }else{  
            throw new Exception("解析的文件格式有误！");
        }  
        return wb;  
    }  
  
    /** 
     * 描述：对表格中数值进行格式化 
     * @param cell 
     * @return 
     */  
    public static Object getCellValue(Cell cell){
        Object value = null;
        DecimalFormat df = new DecimalFormat("0");  //格式化number String字符
        SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd");  //日期格式化
        DecimalFormat df2 = new DecimalFormat("0.00");  //格式化数字
          
        switch (cell.getCellType()) {
        case Cell.CELL_TYPE_STRING:
            value = cell.getRichStringCellValue().getString();  
            break;  
        case Cell.CELL_TYPE_NUMERIC:
            if("General".equals(cell.getCellStyle().getDataFormatString())){  
                value = df.format(cell.getNumericCellValue());  
            }else if("m/d/yy".equals(cell.getCellStyle().getDataFormatString())){  
                value = sdf.format(cell.getDateCellValue());  
            }else{  
                value = df2.format(cell.getNumericCellValue());  
            }  
            break;  
        case Cell.CELL_TYPE_BOOLEAN:
            value = cell.getBooleanCellValue();  
            break;  
        case Cell.CELL_TYPE_BLANK:
            value = "";  
            break;  
        default:  
            break;  
        }  
        return value;  
    }

    public static void main(String[] args) throws FileNotFoundException {
        File file = new File("D:"+File.separator+"工装保养项导入模板.xlsx");
        List<List<Object>> data = getDataListByExcel(new FileInputStream(file), "工装保养项导入模板.xlsx");
        data.forEach(list -> {
            System.out.println(list.get(0));
            System.out.println(list.get(1));
            System.out.println(list.get(2));
            System.out.println(Double.valueOf(list.get(3).toString()));
        });
    }

}
